/*
 * This file is part of LiquidBounce (https://github.com/CCBlueX/LiquidBounce)
 *
 * Copyright (c) 2015 - 2025 CCBlueX
 *
 * LiquidBounce is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * LiquidBounce is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with LiquidBounce. If not, see <https://www.gnu.org/licenses/>.
 */
package net.ccbluex.liquidbounce.features.module.modules.movement

import net.ccbluex.liquidbounce.config.types.nesting.Choice
import net.ccbluex.liquidbounce.config.types.nesting.ChoiceConfigurable
import net.ccbluex.liquidbounce.event.EventState
import net.ccbluex.liquidbounce.event.events.MovementInputEvent
import net.ccbluex.liquidbounce.event.events.PacketEvent
import net.ccbluex.liquidbounce.event.events.PlayerNetworkMovementTickEvent
import net.ccbluex.liquidbounce.event.handler
import net.ccbluex.liquidbounce.features.module.Category
import net.ccbluex.liquidbounce.features.module.ClientModule
import net.ccbluex.liquidbounce.utils.block.collisionShape
import net.ccbluex.liquidbounce.utils.block.getBlock
import net.ccbluex.liquidbounce.utils.client.ceilToInt
import net.ccbluex.liquidbounce.utils.client.floorToInt
import net.ccbluex.liquidbounce.utils.client.sendPacketSilently
import net.ccbluex.liquidbounce.utils.client.sendStartSneaking
import net.ccbluex.liquidbounce.utils.client.sendStopSneaking
import net.ccbluex.liquidbounce.utils.entity.SimulatedPlayer
import net.ccbluex.liquidbounce.utils.entity.copy
import net.ccbluex.liquidbounce.utils.entity.immuneToMagmaBlocks
import net.ccbluex.liquidbounce.utils.entity.moving
import net.ccbluex.liquidbounce.utils.entity.set
import net.ccbluex.liquidbounce.utils.movement.DirectionalInput
import net.minecraft.block.MagmaBlock
import net.minecraft.network.packet.c2s.play.PlayerInputC2SPacket
import net.minecraft.util.math.BlockPos
import net.minecraft.util.math.Box

/**
 * Sneak module
 *
 * Automatically sneaks all the time.
 */
object ModuleSneak : ClientModule("Sneak", Category.MOVEMENT) {

    private val modes = choices("Mode", Vanilla, arrayOf(Legit, Vanilla, Switch)).apply { tagBy(this) }
    private val notDuringMove by boolean("NotDuringMove", false)

    private object Legit : Choice("Legit") {

        private val onMagmaBlocksOnly by boolean("OnMagmaBlocksOnly", false)

        override val parent: ChoiceConfigurable<Choice>
            get() = modes

        @Suppress("unused")
        private val inputHandler = handler<MovementInputEvent> { event ->
            if (player.moving && notDuringMove) {
                return@handler
            }

            if (onMagmaBlocksOnly && (player.immuneToMagmaBlocks || !isOnMagmaBlock(event.directionalInput))) {
                return@handler
            }

            // Temporarily override sneaking
            event.sneak = true
        }

    }

    private object Vanilla : Choice("Vanilla") {

        override val parent: ChoiceConfigurable<Choice>
            get() = modes

        @Suppress("unused")
        private val sneakNetworkHandler = handler<PacketEvent> { event ->
            if ((player.moving && notDuringMove) || event.packet !is PlayerInputC2SPacket) {
                return@handler
            }

            event.cancelEvent() // Because the packet is record
            sendPacketSilently(PlayerInputC2SPacket(event.packet.input.copy(sneak = true)))
        }

    }

    private object Switch : Choice("Switch") {

        var networkSneaking = false

        override val parent: ChoiceConfigurable<Choice>
            get() = modes

        @Suppress("unused")
        val networkTick = handler<PlayerNetworkMovementTickEvent> { event ->
            if (player.moving && notDuringMove) {
                disable()
                return@handler
            }

            when (event.state) {
                EventState.PRE -> {
                    if (networkSneaking) {
                        sendStopSneaking()
                        networkSneaking = false
                    }
                }

                EventState.POST -> {
                    if (!networkSneaking) {
                        sendStartSneaking()
                        networkSneaking = true
                    }
                }
            }
        }

        override fun disable() {
            if (networkSneaking) {
                sendStopSneaking()
                networkSneaking = false
            }
        }
    }

    private fun isOnMagmaBlock(directionalInput: DirectionalInput): Boolean {
        val simulatedInput = SimulatedPlayer.SimulatedPlayerInput.fromClientPlayer(directionalInput)
        simulatedInput.set(jump = false)

        // Doesn't keep the player stuck at the edge of a magma block while sneaking
        simulatedInput.ignoreClippingAtLedge = true

        val simulatedPlayer = SimulatedPlayer.fromClientPlayer(simulatedInput)
        simulatedPlayer.pos = player.pos

        simulatedPlayer.tick()
        val isOnMagmaBlockAfterOneTick = isOnMagmaBlock(simulatedPlayer.boundingBox)

        simulatedPlayer.tick()
        val isOnMagmaBlockAfterTwoTicks = isOnMagmaBlock(simulatedPlayer.boundingBox)

        return isOnMagmaBlockAfterOneTick || isOnMagmaBlockAfterTwoTicks
    }

    /**
     * [boundingBox] - the specific bounding box of a player, mob or even another block.
     */
    private fun isOnMagmaBlock(boundingBox: Box): Boolean {

        // Blocks that are the height of a trapdoor or lower
        // (such as snow layers, carpets, repeaters, or comparators)
        // do not prevent a magma block from damaging mobs and players above it.

        // Therefore, we expand the box downward by 0.2 blocks.
        val expandedBox = boundingBox
            .expand(0.0, 0.1,0.0)
            .offset(0.0, -0.1, 0.0)

        return BlockPos.iterate(
            expandedBox.minX.floorToInt(),
            expandedBox.minY.floorToInt(),
            expandedBox.minZ.floorToInt(),
            expandedBox.maxX.ceilToInt(),
            expandedBox.minY.ceilToInt(),
            expandedBox.maxZ.ceilToInt(),
        ).any {
            it.getBlock() is MagmaBlock &&
                expandedBox.intersects(it.collisionShape.boundingBox.offset(it))
        }
    }
}
